from datetime import datetime
import ollama


def chat():
    now = datetime.now()
    formatted_date = now.strftime('%Y-%m-%d %H:%M')

    system_prompt = f"现在的时间是{formatted_date}, 请帮我把用户提到的时间都转化为 '%Y-%m-%d %H:%M' 格式并返回， 切记不要说明，直接返回结果"

    client = ollama.Client(host='http://localhost:11434')

    def get_response(input_text):
        messages = [
            {
                'role': 'system',
                'content': system_prompt
            },
            {
                'role': 'user',
                'content': input_text
            }
        ]
        response = client.chat(model='llama3.2:latest', messages=messages)
        return response['message']['content']

    print(get_response("明天下午三点20分钟的会议"))
    print(get_response("明天上午8:00的会议"))
    print(get_response("晚上8:00的会议"))
    print(get_response("昨天下午4点的会议"))


if __name__ == '__main__':
    chat()
